package org.demo.util;

import org.dom4j.*;
import org.dom4j.io.SAXReader;

import java.io.IOException;
import java.io.InputStream;
import java.io.StringReader;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.xml.sax.InputSource;

/**
 * @author 别人
 */
public class XmlUtils {

    public static Document xmlTextToDocument(String xmlText){
        try {
            return DocumentHelper.parseText(xmlText);
        } catch (DocumentException e) {
            return null;
        }
    }

    /**
     * xml字符串转换成Map
     * 获取标签内属性值和text值
     *
     * @param xmlText
     * @return
     * @throws Exception
     */
    public static Map<String, String> xmlTextToMap(String xmlText) throws Exception {
        StringReader reader = new StringReader(xmlText);
        InputSource source = new InputSource(reader);
        SAXReader sax = new SAXReader(); // 创建一个SAXReader对象
        Document document = sax.read(source); // 获取document对象,如果文档无节点，则会抛出Exception提前结束
        Element root = document.getRootElement(); // 获取根节点
        Map<String, String> map = getNodes(root); // 从根节点开始遍历所有节点
        return map;
    }

    /**
     * 从指定节点开始,递归遍历所有子节点
     *
     * @author chenleixing
     */
    @SuppressWarnings("unchecked")
    public static Map<String, String> getNodes(Element node) {
        Map<String, String> xmlMap = new HashMap<>();
        xmlMap.put(node.getName().toLowerCase(), node.getTextTrim());
        List<Attribute> listAttr = node.attributes(); // 当前节点的所有属性的list
        for (Attribute attr : listAttr) { // 遍历当前节点的所有属性
            String name = captureName(attr.getName()); // 属性名称 //改为首字母大写
            String value = attr.getValue(); // 属性的值
            xmlMap.put(name, value.trim());
        }


        // 递归遍历当前节点所有的子节点
        List<Element> listElement = node.elements(); // 所有一级子节点的list
        for (Element e : listElement) { // 遍历所有一级子节点
            getNodes(e); // 递归
        }
        return xmlMap;

    }

    //首字母大写
    public static String captureName(String name) {
        name = name.substring(0, 1).toUpperCase() + name.substring(1);
        return name;
    }


    public static Map<String, String> getXmlData(Document document) {
        Map<String, String> map = new HashMap<>(12);
        // 获取根节点
        Element rootElement = document.getRootElement();
        // 获取根节点下所有节点
        List<Element> elements = rootElement.elements();
        for (Element e : elements) {
            // 节点的name为map的key，text为map的value
            map.put(e.getName(), e.getStringValue());
        }
        return map;
    }

    /**
     * 解析xml 获取key-value
     */
    public static Map<String, String> getXmlData(InputStream inputStream) {
        //截取xml
        SAXReader reader = new SAXReader();
        Map<String, String> map = new HashMap<>(12);
        try {
            map = getXmlData(reader.read(inputStream));
        } catch (DocumentException e) {
            e.printStackTrace();
        } finally {
            try {
                inputStream.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return map;
    }

    /**
     * 将xml字符串转为map
     */
    public static Map<String, String> getXmlData(String xmlStr) {
        Map<String, String> map = new HashMap<>(12);
        //截取xml
        try {
            // 将字符串转为XML
            Document document = DocumentHelper.parseText(xmlStr);
            map = getXmlData(document);
        } catch (DocumentException e) {
            e.printStackTrace();
        }
        return map;
    }

}